import datetime

close_fib_file = "D:\\stoc\\lz_strategy\\fib_strategy\\stock_000016_v1.csv"
# close_fib_file = "D:\\stoc\\lz_strategy\\fib_strategy\\stock_399344_v1.csv"
y_column = "return10"
y_scale = 0.05
data_split_rate = 0.7
# 特征值
tezheng = [
    # "quote_rate",
    "amp_rate", # "turnover_rate",
    # "close_ma5", "close_ma20",  ## 不是比例值
    # "turnover_r5",
    "turnover_r20", "turnover_r60",

    # "close_fib_d1", "close_fib_d2", "close_fib_d3",
    # "close_fib_w1", "close_fib_w2", "close_fib_w3",
    # "close_fib_m1", "close_fib_m2", "close_fib_m3",

    # "turnover_fib_d1", "turnover_fib_d2", "turnover_fib_d3",
    # "turnover_fib_w1", "turnover_fib_w2", "turnover_fib_w3",
    # "turnover_fib_m1", "turnover_fib_m2", "turnover_fib_m3",
    # 大盘相关系数
    "corr5", 'corr10', 'corr15', 'corr20', "corr60",
    # 形态
    # "crows2", "crows3",
    # "inside3", "line_strike", "outside",
    # "stars_in_south3", "wihite_soliders3",
    # 动量
    # "adx_14", "adxr_14",#  "apo",
    # "aroon_up", "aroon_down",
    # 'aroon_osc',
    # 'bop',
    # 'cci',
    'cmo', 'dx', 'macd', 'mfi', # 'di',
    "dm", "mom",
    "plus_di", 'plus_dm'
]
tezheng_v1 = []
tezheng_v1.extend(tezheng)
tezheng_v1.extend([
    "close_var",
    "close_var_ma5", "close_var_ma20", "close_var_ma60",
    # "close_var_fib_d1", "close_var_fib_d2", "close_var_fib_d3",
    "close_var_fib_w1", "close_var_fib_w2", "close_var_fib_w3",
    "close_var_fib_m1", "close_var_fib_m2", "close_var_fib_m3"
])

tezheng_v1.extend(["close_ma_fib_w1", "close_ma_fib_m1",
                   "turnover_ma_fib_w1", "turnover_ma_fib_m1",
                   "close_var_ma_fib_w1", "close_var_ma_fib_m1",
                   ])
print(tezheng_v1)
now_time = datetime.datetime.now()
stime = '20201103'
etime = '20201230'

stime1 = datetime.datetime.strptime(stime, '%Y%m%d').strftime('%Y-%m-%d')
etime1 = datetime.datetime.strptime(etime, '%Y%m%d').strftime('%Y-%m-%d')